# Copyright (c) 2018 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""This is all-in-one launch script intended for use by nav2 developers."""

import os

from ament_index_python.packages import get_package_share_directory

from launch import LaunchDescription
from launch.actions import DeclareLaunchArgument, ExecuteProcess, IncludeLaunchDescription
from launch.conditions import IfCondition
from launch.launch_description_sources import PythonLaunchDescriptionSource
from launch.substitutions import LaunchConfiguration, PythonExpression
from launch_ros.actions import Node
#from launch.substitutions import PathJoinSubstitution #为了新增的sitl_dds里面的一些语法
#from launch_ros.substitutions import FindPackageShare  #为了新增的sitl_dds里面的一些语法

def generate_launch_description():
    # Get the launch directory
    bringup_dir = get_package_share_directory('nav2_bringup')
    launch_dir = os.path.join(bringup_dir, 'launch')

    #For new gazebo后加的 增
    #turtlebot3_new_gazebo_dir = get_package_share_directory('turtlebot3_gazebo')
    #turtlebot3_new_gazebo_launch_dir = os.path.join(turtlebot3_new_gazebo_dir,'launch')
    #为了新增的sitl_dds里面的
    #pkg_ardupilot_sitl = get_package_share_directory("ardupilot_sitl")

    # Create the launch configuration variables
    slam = LaunchConfiguration('slam')
    namespace = LaunchConfiguration('namespace')
    use_namespace = LaunchConfiguration('use_namespace')
    map_yaml_file = LaunchConfiguration('map')
    use_sim_time = LaunchConfiguration('use_sim_time')
    params_file = LaunchConfiguration('params_file')
    autostart = LaunchConfiguration('autostart')
    use_composition = LaunchConfiguration('use_composition')
    use_respawn = LaunchConfiguration('use_respawn')

    # Launch configuration variables specific to simulation
    rviz_config_file = LaunchConfiguration('rviz_config_file')
    use_simulator = LaunchConfiguration('use_simulator')
    use_robot_state_pub = LaunchConfiguration('use_robot_state_pub')
    use_rviz = LaunchConfiguration('use_rviz')
    headless = LaunchConfiguration('headless')
    world = LaunchConfiguration('world')
    pose = {'x': LaunchConfiguration('x_pose', default='-2.00'),
            'y': LaunchConfiguration('y_pose', default='-0.50'),
            'z': LaunchConfiguration('z_pose', default='0.01'),
            'R': LaunchConfiguration('roll', default='0.00'),
            'P': LaunchConfiguration('pitch', default='0.00'),
            'Y': LaunchConfiguration('yaw', default='0.00')}
    robot_name = LaunchConfiguration('robot_name')
    robot_sdf = LaunchConfiguration('robot_sdf')

    # Map fully qualified names to relative ones so the node's namespace can be prepended.
    # In case of the transforms (tf), currently, there doesn't seem to be a better alternative
    # https://github.com/ros/geometry2/issues/32
    # https://github.com/ros/robot_state_publisher/pull/30
    # TODO(orduno) Substitute with `PushNodeRemapping`
    #              https://github.com/ros2/launch_ros/issues/56
    remappings = [('/tf', 'tf'),
                  ('/tf_static', 'tf_static')]

    # Declare the launch arguments
    declare_namespace_cmd = DeclareLaunchArgument(
        'namespace',
        default_value='',
        description='Top-level namespace')

    declare_use_namespace_cmd = DeclareLaunchArgument(
        'use_namespace',
        default_value='false',
        description='Whether to apply a namespace to the navigation stack')

    declare_slam_cmd = DeclareLaunchArgument(
        'slam',
        default_value='False',
        description='Whether run a SLAM')

    declare_map_yaml_cmd = DeclareLaunchArgument(
        'map',
        default_value=os.path.join(
            bringup_dir, 'maps', 'turtlebot3_world.yaml'),
        description='Full path to map file to load')

    declare_use_sim_time_cmd = DeclareLaunchArgument(
        'use_sim_time',
        default_value='true',
        description='Use simulation (Gazebo) clock if true')

    declare_params_file_cmd = DeclareLaunchArgument(
        'params_file',
        default_value=os.path.join(bringup_dir, 'params', 'nav2_params.yaml'),
        description='Full path to the ROS2 parameters file to use for all launched nodes')

    declare_autostart_cmd = DeclareLaunchArgument(
        'autostart', default_value='true',
        description='Automatically startup the nav2 stack')

    declare_use_composition_cmd = DeclareLaunchArgument(
        'use_composition', default_value='True',
        description='Whether to use composed bringup')

    declare_use_respawn_cmd = DeclareLaunchArgument(
        'use_respawn', default_value='False',
        description='Whether to respawn if a node crashes. Applied when composition is disabled.')

    declare_rviz_config_file_cmd = DeclareLaunchArgument(
        'rviz_config_file',
        default_value=os.path.join(
            bringup_dir, 'rviz', 'nav2_default_view.rviz'),
        description='Full path to the RVIZ config file to use')

    declare_use_simulator_cmd = DeclareLaunchArgument(
        'use_simulator',
        default_value='True',
        description='Whether to start the simulator')

    declare_use_robot_state_pub_cmd = DeclareLaunchArgument(
        'use_robot_state_pub',
        default_value='True',
        description='Whether to start the robot state publisher')

    declare_use_rviz_cmd = DeclareLaunchArgument(
        'use_rviz',
        default_value='True',
        description='Whether to start RVIZ')

    declare_simulator_cmd = DeclareLaunchArgument(
        'headless',
        default_value='True',
        description='Whether to execute gzclient)')

    declare_world_cmd = DeclareLaunchArgument(
        'world',
        # TODO(orduno) Switch back once ROS argument passing has been fixed upstream
        #              https://github.com/ROBOTIS-GIT/turtlebot3_simulations/issues/91
        # default_value=os.path.join(get_package_share_directory('turtlebot3_gazebo'),
        # worlds/turtlebot3_worlds/waffle.model')
        default_value=os.path.join(bringup_dir, 'worlds', 'world_only.model'),
        description='Full path to world model file to load')

    declare_robot_name_cmd = DeclareLaunchArgument(
        'robot_name',
        default_value='turtlebot3_waffle',
        description='name of the robot')

    declare_robot_sdf_cmd = DeclareLaunchArgument(
        'robot_sdf',
        default_value=os.path.join(bringup_dir, 'worlds', 'waffle.model'),
        description='Full path to robot sdf file to spawn the robot in gazebo')

    # Specify the actions  被注释了
    #start_gazebo_server_cmd = ExecuteProcess(
    #    condition=IfCondition(use_simulator),
    #    cmd=['gzserver', '-s', 'libgazebo_ros_init.so',
    #         '-s', 'libgazebo_ros_factory.so', world],
    #    cwd=[launch_dir], output='screen')

    #启动GUI client 被注释了
    #start_gazebo_client_cmd = ExecuteProcess(
    #    condition=IfCondition(PythonExpression(
    #        [use_simulator, ' and not ', headless])),
    #    cmd=['gzclient'],
    #    cwd=[launch_dir], output='screen')

    #waffle的URDF文件，给robot_state_publisher,之后rviz显示
    urdf = os.path.join(bringup_dir, 'urdf', 'turtlebot3_waffle.urdf')
    with open(urdf, 'r') as infp:
        robot_description = infp.read()

    start_robot_state_publisher_cmd = Node(
        condition=IfCondition(use_robot_state_pub),
        package='robot_state_publisher',
        executable='robot_state_publisher',
        name='robot_state_publisher',
        namespace=namespace,
        output='screen',
        parameters=[{'use_sim_time': use_sim_time,
                     'robot_description': robot_description}],
        remappings=remappings)

    #把SDF文件描述出来，被注释了
    #start_gazebo_spawner_cmd = Node(
    #    package='gazebo_ros',
    #    executable='spawn_entity.py',
    #    output='screen',
    #    arguments=[
    #        '-entity', robot_name,
    #        '-file', robot_sdf,
    #        '-robot_namespace', namespace,
    #        '-x', pose['x'], '-y', pose['y'], '-z', pose['z'],
    #        '-R', pose['R'], '-P', pose['P'], '-Y', pose['Y']])

    #For New gazebo增
    #1.launch gazebo server
    #2.launch gazebo client
    #3.spawn model
#    turtlebot3_new_gazebo_cmd = IncludeLaunchDescription(
#        PythonLaunchDescriptionSource(
#            os.path.join(turtlebot3_new_gazebo_launch_dir,'turtlebot3apm_world.launch.py')),
            #os.path.join(turtlebot3_new_gazebo_launch_dir,'empty_world.launch.py')
    #我认为：模型生成的位置坐标点，此参数将传递给在turtlebot3_world.launch.py/spawn_turtlebot3.launch.py里面的
    # start_gazebo_ros_spawner_cmd节点
    # 节点
#        launch_arguments={
#        'x_pose': pose['x'],
#        'y_pose': pose['y']#,
        #'z_pose': pose['z'],
        #'roll': pose['R'],
        #'pitch': pose['P'],
        #'yaw': pose['Y']
#        }.items())
 
    # 新增，启动了六轮车的sitl
#    sitl_dds = IncludeLaunchDescription(
#        PythonLaunchDescriptionSource(
#           [
#                PathJoinSubstitution(
#                    [
#                        FindPackageShare("ardupilot_sitl"),
#                        "launch",
#                        "sitl_dds_udp.launch.py",
#                    ]
#                ),
#            ]
#        ),
#        launch_arguments={
#            "transport": "udp4",
#            "port": "2019",
#            "command": "ardurover",
#            "synthetic_clock": "True",
#            "wipe": "False",
#            "model": "json",
#            "speedup": "1",
#            "slave": "0",
#            "instance": "0",
#            "defaults": os.path.join(
#                pkg_ardupilot_sitl,
#                "config",
#                "default_params",
#                "rover-skid.parm",
#            )
#            + ","
#            + os.path.join(
#                pkg_ardupilot_sitl,
#                "config",
#                "default_params",
#                "dds_udp.parm",
#            ),
#            "sim_address": "127.0.0.1",
#            "master": "tcp:127.0.0.1:5760",
#            "sitl": "127.0.0.1:5501",
#        }.items(),
#    )


    rviz_cmd = IncludeLaunchDescription(
        PythonLaunchDescriptionSource(
            os.path.join(launch_dir, 'rviz_launch.py')),
        condition=IfCondition(use_rviz),
        launch_arguments={'namespace': namespace,
                          'use_namespace': use_namespace,
                          'rviz_config': rviz_config_file}.items())

    bringup_cmd = IncludeLaunchDescription(
        PythonLaunchDescriptionSource(
            os.path.join(launch_dir, 'bringup_launch.py')),
        launch_arguments={'namespace': namespace,
                          'use_namespace': use_namespace,
                          'slam': slam,
                          'map': map_yaml_file,
                          'use_sim_time': use_sim_time,
                          'params_file': params_file,
                          'autostart': autostart,
                          'use_composition': use_composition,
                          'use_respawn': use_respawn}.items())

    # Create the launch description and populate
    ld = LaunchDescription()

    # Declare the launch options
    ld.add_action(declare_namespace_cmd)
    ld.add_action(declare_use_namespace_cmd)
    ld.add_action(declare_slam_cmd)
    ld.add_action(declare_map_yaml_cmd)
    ld.add_action(declare_use_sim_time_cmd)
    ld.add_action(declare_params_file_cmd)
    ld.add_action(declare_autostart_cmd)
    ld.add_action(declare_use_composition_cmd)

    ld.add_action(declare_rviz_config_file_cmd)
    ld.add_action(declare_use_simulator_cmd)
    ld.add_action(declare_use_robot_state_pub_cmd)
    ld.add_action(declare_use_rviz_cmd)
    ld.add_action(declare_simulator_cmd)
    ld.add_action(declare_world_cmd)
    ld.add_action(declare_robot_name_cmd)
    ld.add_action(declare_robot_sdf_cmd)
    ld.add_action(declare_use_respawn_cmd)

    # Add any conditioned actions
    #ld.add_action(start_gazebo_server_cmd)
    #ld.add_action(start_gazebo_client_cmd)
    #ld.add_action(start_gazebo_spawner_cmd)

    #For new gazebo 增
    #ld.add_action(turtlebot3_new_gazebo_cmd)
    #启动六轮车的SITL
    #ld.add_action(sitl_dds)                
    
    # Add the actions to launch all of the navigation nodes
    ld.add_action(start_robot_state_publisher_cmd)
    ld.add_action(rviz_cmd)
    ld.add_action(bringup_cmd)

    return ld
